Solution in Python